Rethinking Cross-Tokenizer On-Policy Distillation: From Alignment Coverage to Supervision Reliability
跨分词器在线策略蒸馏再思考:从对齐覆盖到监督可靠性
🧭 跨 tokenizer 蒸馏优化 ★ 深度解读
Alibaba Cloud Computing
跨 tokenizer 在线蒸馏中,严格 1:1 对齐已覆盖多数 token;学生选 top-16 共享词表即可保留大部分收益,补 mismatch span MSE 反而降精度
📌 工作参考价值(4 点)
- 做跨 tokenizer / 跨模型蒸馏时,可先只对严格 1:1 对齐位置做 shared vocabulary 上的 reverse KL,并让学生自选 top-k(如 top-16)词表子集;这能压缩计算和显存,同时保留大部分蒸馏收益。适合生成式推荐中不同 tokenizer 的 Semantic ID 序列蒸馏。
- 新增辅助监督(如 mismatch span MSE、多任务损失)前,先做梯度方向与相对幅度诊断:计算 g_mis 与 g_strict 的 cosine 和 norm ratio;若 cosine 接近 0/负、norm ratio 随训练上升,大概率引入冲突信号,应降低权重或丢弃。
- 评估“监督覆盖”不要只看静态词表 Jaccard,要看 student rollout 上 strict token coverage 以及共享词表实际概率质量;工程上采样几十条 rollout 即可快速判断是否值得做复杂对齐。
- 在 Agent 或推荐 LLM 的蒸馏中,若希望提升 teacher 信号覆盖,不要盲目追求全位置覆盖;优先保证主损失梯度干净,可先在小支持集上验证收益再扩展。